#!/bin/bash
APP=$@

local_rank=$OMPI_COMM_WORLD_LOCAL_RANK

case ${local_rank} in
[0])
 export HIP_VISIBLE_DEVICES=0
  $APP
  ;;
[1])
 export HIP_VISIBLE_DEVICES=1
  $APP
  ;;
esac
